package com.wuxuan.algorithm_homework.array;

import java.util.Arrays;
import java.util.function.Consumer;

/**
 * 动态数组
 *
 * @param <E>
 */
public class MyArrayList<E> {
    private int size;
    private E[] arrayList;
    private final static int DEFAULT_CAPACITY = 10;

    /**
     * 初始化
     */
    public MyArrayList() {
        size = 0;
        //类似  int[] arr = new int[10];
        //Java 不允许 new E[...]，因为泛型在运行时被擦除。
        //所以我这里先 new Object[...]然后再强转为E[]类型
        arrayList = (E[]) new Object[DEFAULT_CAPACITY];
    }


    public int size() {
        return size;
    }

    /**
     * 在最后的位置添加元素
     * @param e
     */
    public void add(E e) {
        if (size == arrayList.length) {
            //扩容
            grow();
        }
        arrayList[size++] = e;
    }


    /**
     * 修改/设置元素
     * @param index
     * @param e
     */
    public boolean update(int index, E e) {
        if (size == 0 || index < 0 || index >= size) {
            return false;
        }
        arrayList[index] = e;
        return true;
    }

    /**
     * 删除元素
     * @param index
     * @return
     */
    public E remove(int index) {
        if (size == 0 || index < 0 || index >= size) {
            return null;
        }
        E e = arrayList[index];
        //删除数据应该要考虑是否需要进行元素移位操作
        int moveFlag = size - index - 1;
        if (moveFlag != 0){
            //把数据进行移位操作
            System.arraycopy(arrayList, index + 1, arrayList, index, moveFlag);
        }
        //把当前最后一个数据改成null
        arrayList[--size] = null;
        return e;
    }

    /**
     * 获取元素
     * @param index
     * @return
     */
    public E get(int index) {
        if (size == 0 || index < 0 || index >= size) {
            return null;
        }
        return arrayList[index];
    }

    /**
     * 扩容
     */
    public void grow() {
        //扩为原来1.5倍
        arrayList = Arrays.copyOf(arrayList, arrayList.length + arrayList.length >> 1);
    }

    /**
     * 重写toString，仿照数组的返回形式
     * @return
     */
    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append("[");
        for (int i = 0; i < size; i++) {
            E e = arrayList[i];
            sb.append(e).append(i == size - 1 ? "" : ", ");
        }
        sb.append("]");
        return sb.toString();
    }

}
